package com.ruoyi.water.controller;

import java.util.List;
import java.util.Date;
import jakarta.servlet.http.HttpServletResponse;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.PutMapping;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.format.annotation.DateTimeFormat;
import com.ruoyi.common.annotation.Log;
import com.ruoyi.common.core.controller.BaseController;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.core.page.TableDataInfo;
import com.ruoyi.common.enums.BusinessType;
import com.ruoyi.water.domain.AreaAnalyse;
import com.ruoyi.water.service.IAreaAnalyseService;
import com.ruoyi.common.utils.poi.ExcelUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;

/**
 * 区域分析结果Controller
 *
 * @author wujiong
 * @date 2025-09-23
 */
@RestController
@RequestMapping("/water/areaAnalyse")
public class AreaAnalyseController extends BaseController
{
    @Autowired
    private IAreaAnalyseService areaAnalyseService;

    /**
     * 查询区域分析结果列表（兼容：单时间精确匹配 + 时间范围筛选）
     */
    @PreAuthorize("@ss.hasPermi('water:areaAnalyse:list')")
    @GetMapping("/list")
    public TableDataInfo list(
            AreaAnalyse areaAnalyse,
            @RequestParam(required = false) String startTime, // 新：时间范围-开始（yyyy-MM-dd）
            @RequestParam(required = false) String endTime     // 新：时间范围-结束（yyyy-MM-dd）
    ) {
        // 1. 统一分页处理（复用BaseController内置方法，自动读pageNum/pageSize，默认1/10）
        startPage();

        // 2. 构建查询条件：优先精确匹配，再处理范围筛选
        QueryWrapper<AreaAnalyse> queryWrapper = new QueryWrapper<>();

        // 原有条件：ID、区域ID筛选（保留不变）
        if (areaAnalyse.getId() != null) {
            queryWrapper.eq("id", areaAnalyse.getId());
        }
        if (areaAnalyse.getAreaId() != null) {
            queryWrapper.eq("areaId", areaAnalyse.getAreaId());
        }

        // 核心：时间条件兼容逻辑（精确匹配 > 范围筛选）
        if (areaAnalyse.getAnalyseDate() != null) {
            // 场景1：传入了analyseDate（原单时间条件），优先精确匹配
            queryWrapper.eq("analyseDate", areaAnalyse.getAnalyseDate());
        } else {
            // 场景2：未传analyseDate，判断是否用时间范围筛选
            if (startTime != null && !startTime.trim().isEmpty()) {
                queryWrapper.ge("analyseDate", startTime); // 大于等于开始时间
            }
            if (endTime != null && !endTime.trim().isEmpty()) {
                // 若传了endTime，先校验开始时间不晚于结束时间（避免无效范围）
                if (startTime != null && !startTime.trim().isEmpty()) {
                    if (startTime.compareTo(endTime) > 0) {
                        throw new RuntimeException("开始时间不能晚于结束时间"); // 抛异常触发统一异常处理
                    }
                }
                queryWrapper.le("analyseDate", endTime); // 小于等于结束时间
            }
        }

        // 原有排序：按分析日期升序（与MeterAnalyseController对齐，保留）
        queryWrapper.orderByAsc("analyseDate");

        // 3. 执行查询（startPage()已拦截分页，list()自动返回分页数据）
        List<AreaAnalyse> list = areaAnalyseService.list(queryWrapper);

        // 4. 统一返回TableDataInfo（包含分页全量信息，前端无需适配新格式）
        return getDataTable(list);
    }

    // ------------------------------ 以下方法完全保留不变 ------------------------------
    /**
     * 计算指定日期范围内的区域日流量
     */
    @PreAuthorize("@ss.hasPermi('water:areaAnalyse:process')")
    @Log(title = "区域分析结果", businessType = BusinessType.OTHER)
    @PostMapping("/process")
    public AjaxResult calculateDailyFlow(
            @RequestParam @DateTimeFormat(pattern = "yyyy-MM-dd") Date startDate,
            @RequestParam @DateTimeFormat(pattern = "yyyy-MM-dd") Date endDate) {

        if (startDate == null || endDate == null) {
            return AjaxResult.error("请传入有效的日期范围");
        }
        if (startDate.after(endDate)) {
            return AjaxResult.error("开始日期不能晚于结束日期");
        }

        List<AreaAnalyse> result = areaAnalyseService.calculateAreaDailyFlow(startDate, endDate);
        return AjaxResult.success("计算完成", result);
    }

    /**
     * 计算指定日期范围内的区域偏差数据
     */
    @PreAuthorize("@ss.hasPermi('water:areaAnalyse:process')")
    @Log(title = "区域分析结果", businessType = BusinessType.OTHER)
    @PostMapping("/processBias")
    public AjaxResult calculateBiasData(
            @RequestParam @DateTimeFormat(pattern = "yyyy-MM-dd") Date startDate,
            @RequestParam @DateTimeFormat(pattern = "yyyy-MM-dd") Date endDate) {

        if (startDate == null || endDate == null) {
            return AjaxResult.error("请传入有效的日期范围");
        }
        if (startDate.after(endDate)) {
            return AjaxResult.error("开始日期不能晚于结束日期");
        }

        List<AreaAnalyse> result = areaAnalyseService.calculateAreaBiasData(startDate, endDate);
        return AjaxResult.success("偏差数据计算完成", result);
    }

    /**
     * 导出区域分析结果列表
     */
    @PreAuthorize("@ss.hasPermi('water:areaAnalyse:export')")
    @Log(title = "区域分析结果", businessType = BusinessType.EXPORT)
    @PostMapping("/export")
    public void export(HttpServletResponse response, AreaAnalyse areaAnalyse)
    {
        QueryWrapper<AreaAnalyse> queryWrapper = new QueryWrapper<>(areaAnalyse);
        List<AreaAnalyse> list = areaAnalyseService.list(queryWrapper);
        ExcelUtil<AreaAnalyse> util = new ExcelUtil<AreaAnalyse>(AreaAnalyse.class);
        util.exportExcel(response, list, "区域分析结果数据");
    }

    /**
     * 获取区域分析结果详细信息
     */
    @PreAuthorize("@ss.hasPermi('water:areaAnalyse:query')")
    @GetMapping(value = "/{id}")
    public AjaxResult getInfo(@PathVariable("id") Long id)
    {
        return success(areaAnalyseService.getById(id));
    }

    /**
     * 新增区域分析结果
     */
    @PreAuthorize("@ss.hasPermi('water:areaAnalyse:add')")
    @Log(title = "区域分析结果", businessType = BusinessType.INSERT)
    @PostMapping
    public AjaxResult add(@RequestBody AreaAnalyse areaAnalyse)
    {
        return toAjax(areaAnalyseService.save(areaAnalyse) ? 1 : 0);
    }

    /**
     * 修改区域分析结果
     */
    @PreAuthorize("@ss.hasPermi('water:areaAnalyse:edit')")
    @Log(title = "区域分析结果", businessType = BusinessType.UPDATE)
    @PutMapping
    public AjaxResult edit(@RequestBody AreaAnalyse areaAnalyse)
    {
        return toAjax(areaAnalyseService.updateById(areaAnalyse) ? 1 : 0);
    }

    /**
     * 删除区域分析结果
     */
    @PreAuthorize("@ss.hasPermi('water:areaAnalyse:remove')")
    @Log(title = "区域分析结果", businessType = BusinessType.DELETE)
    @DeleteMapping("/{ids}")
    public AjaxResult remove(@PathVariable Long[] ids)
    {
        return toAjax(areaAnalyseService.removeByIds(List.of(ids)) ? ids.length : 0);
    }
}